package JZOffer;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class Jz32III {

    // Jz.32III.从上到下打印二叉树
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }

    /**
     * 使用广度优先搜索的方法完成树的遍历
     * @param root
     * @return List<List<Integer>>
     */
    public List<List<Integer>> levelOrder(TreeNode root) {
        // 二维数组
        List<List<Integer>> lists = new ArrayList<List<Integer>>();
        // 新建一个用链实现的队列用于存储节点
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        // 判断是否为空二叉树
        if(root != null) {
            queue.add(root);
        }
        while (!queue.isEmpty()) {
            //创建arrayList用于存储每一层节点
            LinkedList<Integer> list = new LinkedList<>();
            for (int i = queue.size(); i > 0 ; i--) {
                TreeNode node = queue.poll();
//                list.add(node.val);
                if(lists.size() % 2 == 0) list.addLast(node.val); // 偶数层 -> 队列头部
                else list.addFirst(node.val); // 奇数层 -> 队列尾部
                // 如果左节点不为空入列
                if (node.left != null) {
                    queue.offer(node.left);
                }
                // 如果右节点不为空入列
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            lists.add(list);
        }
        return lists;
    }

}
